"""
Train AutoEncoder-KL
"""
